#!/bin/bash

# 获取脚本所在目录
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$SCRIPT_DIR"

echo "正在启动抖音热搜选题工具..."
echo

# 激活虚拟环境
if [ -d "venv" ]; then
    echo "激活虚拟环境..."
    source venv/bin/activate
    
    # 检查虚拟环境是否激活成功
    if [ $? -eq 0 ]; then
        echo "虚拟环境已激活"
        echo "正在启动 Streamlit 应用..."
        echo
        
        # 启动 Streamlit 应用
        streamlit run main.py
    else
        echo "错误：无法激活虚拟环境"
        read -p "按任意键退出..."
        exit 1
    fi
else
    echo "错误：未找到虚拟环境目录 'venv'"
    echo "请确保在项目根目录下运行此脚本"
    read -p "按任意键退出..."
    exit 1
fi

# 如果 Streamlit 退出，暂停以便查看错误信息
if [ $? -ne 0 ]; then
    echo
    echo "应用启动失败，请检查错误信息"
    read -p "按任意键退出..."
fi